package com.cloudpivot.receivales;

import com.cloudpivot.receivales.ding.robot.domain.ErnieMessageDTO;
import com.cloudpivot.receivales.ding.robot.domain.ErnieRequestDTO;
import com.cloudpivot.receivales.ding.robot.service.baidu.BaiduERNIEServiceImpl;
import com.cloudpivot.receivales.ding.robot.service.baidu.BaiduNPLServiceImpl;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;

import java.lang.reflect.Array;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

@SpringBootTest
class CloudpivotApplicationForBaidu {

    @Autowired
    private BaiduNPLServiceImpl baiduService;

    @Autowired
    private BaiduERNIEServiceImpl baiduERNIEService;

    @Test
    public void baiduSimForTest() {
        String one = "安全";
        String two = "上网";
        baiduService.wordEmbSimServiceForRest(one, two);

    }

    @Test
    public void baiduSimForTest2() {
        String one = "博圣安全上网操作指南";
        String two = "安全";
        baiduService.embSimService(one, two);

    }

    @Test
    public void ernieTest1(){
        ErnieMessageDTO message1 = new ErnieMessageDTO();
        ErnieMessageDTO message2 = new ErnieMessageDTO();
        ErnieMessageDTO message3 = new ErnieMessageDTO();
        message1.setRole("user");
        message1.setContent("你好");
        message2.setRole("assistant");
        message2.setContent("如果有任何问题可随时像我提问");
        message3.setRole("user");
        message3.setContent("请对以下内容进行润色，使其更加清晰、简洁、有逻辑. 原文：本周内容:20240515 访客计划：孝感市妇幼保健院招标事宜 跟进记录：外送招标沟通，需求文件协同修改 " +
                "20240515 访客计划：孝感市妇幼保健院外送招标，云梦，安陆区县串联宣教，区县开发拜访 跟进记录：串联区县上量宣教及信息收集" +
                "20240516 访客计划：孝感市妇幼保健院议价流程，对账后续 跟进记录：试剂议价完成，平台授权跟进后签订合同，会议协助 " +
                "20240518 访客计划：孝感市妇幼保健院外送招标及回款跟进 跟进记录：招标公司协调，外送项目沟通 要求:1. 尽量保留原文中的关键信息，如访客计划、跟进记录等。" +
                "2. 按照如下结构输出文件:本周已完成工作,本周推进中的工作,需要再例会中讨论的内容,下周重点工作");
        List<ErnieMessageDTO> messages = new ArrayList<>();
        messages.add(message1);
        messages.add(message2);
        messages.add(message3);
        ErnieRequestDTO requestDTO = new ErnieRequestDTO();
        requestDTO.setMessages(messages);
        baiduERNIEService.ernie4PreView(messages);
    }
}
